package com.cnd.scm.datasource.util;

import com.cnd.scm.datasource.constant.DataSourceKey;

/**
 * @version V1.0
 * @ClassName: DataSourceHolder
 * @Description:TODO(用于数据源切换)
 * @author: 李启联
 * @date: 2019-12-18 11:06
 * @Copyright: Copyright (c), 2018-2019
 * 申明：本内容未经作者本人同意,禁止外泄以及用于其他的商业项目.
 */
public class DataSourceHolder {
    //注意使用ThreadLocal，微服务下游建议使用信号量
    private static final ThreadLocal<DataSourceKey> dataSourceKey = new ThreadLocal<>();

    //得到当前的数据库连接
    public static DataSourceKey getDataSourceKey() {
        return dataSourceKey.get();
    }
    //设置当前的数据库连接
    public static void setDataSourceKey(DataSourceKey type) {
        dataSourceKey.set(type);
    }
    //清除当前的数据库连接
    public static void clearDataSourceKey() {
        dataSourceKey.remove();
    }
}
